import logging
from logging.handlers import RotatingFileHandler
import os
def setup_logging():
    # 确保日志目录存在
    LOG_DIR = "logs"
    if not os.path.exists(LOG_DIR):
        os.makedirs(LOG_DIR)
    # 配置日志格式
    log_formatter = logging.Formatter(
        "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
    )
    # 设置RotatingFileHandler，当文件达到一定大小后自动轮转
    file_handler = RotatingFileHandler(
        f"{LOG_DIR}/app.log",
        maxBytes=1024 * 1024 * 10,  # 10MB
        backupCount=10,  # 最多保留10个备份文件
        encoding="utf-8"
    )
    file_handler.setFormatter(log_formatter)
    file_handler.setLevel(logging.INFO)
    # 控制台日志处理器
    console_handler = logging.StreamHandler()
    console_handler.setFormatter(log_formatter)
    console_handler.setLevel(logging.INFO)
    # 配置根日志
    logging.basicConfig(
        level=logging.INFO,
        handlers=[file_handler, console_handler]
    )
    # 返回专用日志器
    return logging.getLogger("fastapi_app")
